import sys
import torch

in_chkpt = sys.argv[1]
out_chkpt= sys.argv[2]
keys = sys.argv[3]
value = sys.argv[4]


